package com.sann.demo;

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.GlobalConfig;
import com.baomidou.mybatisplus.generator.config.PackageConfig;
import com.baomidou.mybatisplus.generator.config.StrategyConfig;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import lombok.extern.slf4j.Slf4j;

/**
 * 代码生成器：根据数据库表，生成对表的基本操作代码。
 */
@Slf4j
public class CodeGenerator {

    /**
     * 执行逆向生成代码
     *
     * 需要修改部分配置信息，符合你的项目需要
     *
     */
    public static void main(String[] args) {
        executeGenerator();
    }

    public static void executeGenerator(){
        // 1、创建代码生成器
        AutoGenerator mpg = new AutoGenerator();

        // 2、全局配置
        GlobalConfig gc = new GlobalConfig();
//        String projectPath = System.getProperty("user.dir");
//        log.info(projectPath); //E:\ws\guli_edu\guli_parent
        // 修改：修改输出目录
        gc.setOutputDir("E:\\ws\\guli_edu\\guli_parent\\service\\service_edu" + "\\src\\main\\java");

        gc.setAuthor("zjw");
        gc.setOpen(false); //生成后是否打开资源管理器
        gc.setFileOverride(false); //重新生成时文件是否覆盖,true:表示重新生成的代码会覆盖原有的代码

        //IUserService -> UserService
        gc.setServiceName("%sService");	//去掉Service接口的首字母I

        // 修改：根据实际数据库主键是Number(Long/Integer)或String请使用ASSIGN_ID，如果是UUID请使用ASSIGN_UUID
        gc.setIdType(IdType.ASSIGN_ID); //主键策略
        gc.setDateType(DateType.ONLY_DATE);//定义生成的实体类中日期类型
        gc.setSwagger2(true);//开启Swagger2模式

        mpg.setGlobalConfig(gc);

        // 3、数据源配置
        DataSourceConfig dsc = new DataSourceConfig();
        dsc.setUrl("jdbc:mysql://localhost:3306/guli_edu?useUnicode=true&useSSL=false&characterEncoding=utf8&serverTimezone=GMT%2B8");
        dsc.setDriverName("com.mysql.cj.jdbc.Driver");
        dsc.setUsername("root");
        dsc.setPassword("000000");
        dsc.setDbType(DbType.MYSQL);
        mpg.setDataSource(dsc);

        // 4、包配置
        PackageConfig pc = new PackageConfig();
        pc.setParent("com.sann");
        //包  com.sann.eduservice
        pc.setModuleName("eduservice"); //模块名
        //包  com.sann.eduservice.controller
        pc.setController("controller");
        pc.setEntity("entity");
        pc.setService("service");
        pc.setMapper("mapper");
        mpg.setPackageInfo(pc);

        // 5、策略配置
        StrategyConfig strategy = new StrategyConfig();
        strategy.setInclude("edu_course","edu_chapter","edu_course_description","edu_video");//表名，多张表可以逗号隔开
        strategy.setNaming(NamingStrategy.underline_to_camel);//数据库表映射到实体的命名策略
//        strategy.setTablePrefix(pc.getModuleName() + "_"); //生成实体时去掉表前缀
        strategy.setTablePrefix("edu_"); //生成实体时去掉表前缀

        strategy.setColumnNaming(NamingStrategy.underline_to_camel);//数据库表字段映射到实体的命名策略
        strategy.setEntityLombokModel(true); // lombok 模型 @Accessors(chain = true) setter链式操作

        strategy.setRestControllerStyle(true); //restful api风格控制器
        strategy.setControllerMappingHyphenStyle(true); //url中驼峰转连字符

//        strategy.setSuperEntityClass("你自己的父类实体,没有就不用设置");


        mpg.setStrategy(strategy);

        // 6、执行
        mpg.execute();
    }
}
